import torch
from torch import nn

# 1. 定义单个层
hidden_layer = nn.Linear(in_features=20, out_features=256)  # 定义隐藏层
activation_layer = nn.ReLU()                                # 定义激活层
output_layer = nn.Linear(in_features=256, out_features=10)  # 定义输出层

# 2. 使用nn.Sequential定义神经网络块
net = nn.Sequential(
    hidden_layer,    # 第一个层：全连接隐藏层
    activation_layer, # 第二个层：ReLU激活函数
    output_layer     # 第三个层：全连接输出层
)
print(net)
